Node / meshtastic / Meshtastic-Android / files / core / database / src / commonTest / kotlin / org / meshtastic / core / database / dao / SwitchingDiscoveryDaoTest.kt

Displaying Raw • Download

T8b949e/*

* Copyright (c) 2026 Meshtastic LLC

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <https://www.gnu.org/licenses/>.

*/

Tff7b72package T7ee787org.meshtastic.core.database.dao

Tff7b72import T7ee787kotlinx.coroutines.ExperimentalCoroutinesApi

Tff7b72import T7ee787kotlinx.coroutines.flow.Flow

Tff7b72import T7ee787kotlinx.coroutines.flow.MutableStateFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.StateFlow

Tff7b72import T7ee787kotlinx.coroutines.flow.first

Tff7b72import T7ee787kotlinx.coroutines.flow.flatMapLatest

Tff7b72import T7ee787kotlinx.coroutines.flow.onEach

Tff7b72import T7ee787kotlinx.coroutines.test.runTest

Tff7b72import T7ee787org.meshtastic.core.database.DatabaseProvider

Tff7b72import T7ee787org.meshtastic.core.database.MeshtasticDatabase

Tff7b72import T7ee787org.meshtastic.core.database.entity.DiscoverySessionEntity

Tff7b72import T7ee787org.meshtastic.core.database.entity.DiscoverySessionStatus

Tff7b72import T7ee787org.meshtastic.core.database.getInMemoryDatabaseBuilder

Tff7b72import T7ee787kotlin.test.AfterTest

Tff7b72import T7ee787kotlin.test.Test

Tff7b72import T7ee787kotlin.test.assertEquals

Tff7b72import T7ee787kotlin.test.assertTrue

T8b949e/**

* Regression for #6235: the DAO Koin hands to `feature:discovery` used to be pinned to the injection-time database, so

* a device/DB switch left long-lived consumers reading (and writing!) the old DB. [SwitchingDiscoveryDao] must resolve

* the active database per call/collection instead.

*/

Tff7b72class T56d364SwitchingDiscoveryDaoTest Tb4b4b4{

Tff7b72private Tff7b72val Te6edf3dbATb4b4b4: Te6edf3MeshtasticDatabase Tff7b72= Te6edf3getInMemoryDatabaseBuilderTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Tff7b72private Tff7b72val Te6edf3dbBTb4b4b4: Te6edf3MeshtasticDatabase Tff7b72= Te6edf3getInMemoryDatabaseBuilderTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Tff7b72private Tff7b72val Te6edf3provider Tff7b72= Te6edf3TestProviderTb4b4b4(Te6edf3dbATb4b4b4)

Tff7b72private Tff7b72val Te6edf3dao Tff7b72= Te6edf3SwitchingDiscoveryDaoTb4b4b4(Te6edf3providerTb4b4b4)

Tf0883e@AfterTest

Tff7b72fun Td2a8fftearDownTb4b4b4(Tb4b4b4) Tb4b4b4{

Te6edf3dbATb4b4b4.Te6edf3closeTb4b4b4(Tb4b4b4)

Te6edf3dbBTb4b4b4.Te6edf3closeTb4b4b4(Tb4b4b4)

Tb4b4b4}

Tff7b72private Tff7b72fun Td2a8ffsessionTb4b4b4(Te6edf3timestampTb4b4b4: Tffa657LongTb4b4b4) Tff7b72=

Te6edf3DiscoverySessionEntityTb4b4b4(Te6edf3timestamp Tff7b72= Te6edf3timestampTb4b4b4, Te6edf3presetsScanned Tff7b72= Ta5d6ff"Ta5d6ffLongFastTa5d6ff"Tb4b4b4, Te6edf3homePreset Tff7b72= Ta5d6ff"Ta5d6ffLongFastTa5d6ff"Tb4b4b4)

Tf0883e@Test

Tff7b72fun Td2a8ffsuspendCallsResolveTheCurrentDbPerCallTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Te6edf3daoTb4b4b4.Te6edf3insertSessionTb4b4b4(Te6edf3sessionTb4b4b4(Te6edf3timestamp Tff7b72= T79c0ff1Tb4b4b4)Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3dbATb4b4b4.Te6edf3discoveryDaoTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3getAllSessionsSnapshotTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3sizeTb4b4b4, Ta5d6ff"Ta5d6fffirst insert lands in the active DB (A)Ta5d6ff"Tb4b4b4)

Te6edf3providerTb4b4b4.Te6edf3switchToTb4b4b4(Te6edf3dbBTb4b4b4)

Te6edf3daoTb4b4b4.Te6edf3insertSessionTb4b4b4(Te6edf3sessionTb4b4b4(Te6edf3timestamp Tff7b72= T79c0ff2Tb4b4b4)Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3dbATb4b4b4.Te6edf3discoveryDaoTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3getAllSessionsSnapshotTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3sizeTb4b4b4, Ta5d6ff"Ta5d6ffold DB untouched after switchTa5d6ff"Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3dbBTb4b4b4.Te6edf3discoveryDaoTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3getAllSessionsSnapshotTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3sizeTb4b4b4, Ta5d6ff"Ta5d6ffpost-switch insert lands in the new DB (B)Ta5d6ff"Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff1Tb4b4b4, Te6edf3daoTb4b4b4.Te6edf3getAllSessionsSnapshotTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3sizeTb4b4b4, Ta5d6ff"Ta5d6ffreads resolve the new DB tooTa5d6ff"Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffstatusUpdateReportsUnavailableDatabaseTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3sessionId Tff7b72=

Te6edf3daoTb4b4b4.Te6edf3insertSessionTb4b4b4(Te6edf3sessionTb4b4b4(Te6edf3timestamp Tff7b72= T79c0ff1Tb4b4b4)Tb4b4b4.Te6edf3copyTb4b4b4(Te6edf3completionStatus Tff7b72= Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3IN_PROGRESSTb4b4b4)Tb4b4b4)

Te6edf3providerTb4b4b4.Te6edf3setWritesAvailableTb4b4b4(Tff7b72falseTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff0Tb4b4b4, Te6edf3daoTb4b4b4.Te6edf3updateSessionCompletionStatusTb4b4b4(Te6edf3sessionIdTb4b4b4, Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3FAILEDTb4b4b4)Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(

Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3IN_PROGRESSTb4b4b4,

Te6edf3dbATb4b4b4.Te6edf3discoveryDaoTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3getSessionTb4b4b4(Te6edf3sessionIdTb4b4b4)Tff7b72?.Te6edf3completionStatusTb4b4b4,

Ta5d6ff"Ta5d6ffan unavailable database must not be reported as a successful status writeTa5d6ff"Tb4b4b4,

Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffrecoverableStatusUpdateReportsUnavailableDatabaseTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3sessionId Tff7b72=

Te6edf3daoTb4b4b4.Te6edf3insertSessionTb4b4b4(Te6edf3sessionTb4b4b4(Te6edf3timestamp Tff7b72= T79c0ff1Tb4b4b4)Tb4b4b4.Te6edf3copyTb4b4b4(Te6edf3completionStatus Tff7b72= Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3IN_PROGRESSTb4b4b4)Tb4b4b4)

Te6edf3providerTb4b4b4.Te6edf3setWritesAvailableTb4b4b4(Tff7b72falseTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff0Tb4b4b4, Te6edf3daoTb4b4b4.Te6edf3updateRecoverableSessionCompletionStatusTb4b4b4(Te6edf3sessionIdTb4b4b4, Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3FAILEDTb4b4b4)Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(

Te6edf3DiscoverySessionStatusTb4b4b4.Te6edf3IN_PROGRESSTb4b4b4,

Te6edf3dbATb4b4b4.Te6edf3discoveryDaoTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3getSessionTb4b4b4(Te6edf3sessionIdTb4b4b4)Tff7b72?.Te6edf3completionStatusTb4b4b4,

Ta5d6ff"Ta5d6ffan unavailable database must not be reported as a successful recoverable status writeTa5d6ff"Tb4b4b4,

Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffflowsRelatchOntoTheCurrentDbTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Te6edf3daoTb4b4b4.Te6edf3insertSessionTb4b4b4(Te6edf3sessionTb4b4b4(Te6edf3timestamp Tff7b72= T79c0ff1Tb4b4b4)Tb4b4b4)

T8b949e// One live collection end-to-end: the first emission (DB A's session) triggers the switch, and the SAME

T8b949e// collector must then receive DB B's empty state. A delegate that resolved currentDb only at flow creation

T8b949e// would never emit the empty list, and the test would fail on runTest's timeout.

Tff7b72var Te6edf3sawDbA Tff7b72= Tff7b72false

Tff7b72val Te6edf3relatched Tff7b72=

Te6edf3daoTb4b4b4.Te6edf3getAllSessionsTb4b4b4(Tb4b4b4)

Tb4b4b4.Te6edf3onEach Tb4b4b4{ Te6edf3sessions Tff7b72-Tff7b72>

Tff7b72if Tb4b4b4(Te6edf3sessionsTb4b4b4.Te6edf3isNotEmptyTb4b4b4(Tb4b4b4)Tb4b4b4) Tb4b4b4{

Te6edf3sawDbA Tff7b72= Tff7b72true

Te6edf3providerTb4b4b4.Te6edf3switchToTb4b4b4(Te6edf3dbBTb4b4b4)

Tb4b4b4}

Tb4b4b4}

Tb4b4b4.Te6edf3first Tb4b4b4{ Tffa657itTb4b4b4.Te6edf3isEmptyTb4b4b4(Tb4b4b4) Tb4b4b4}

Te6edf3assertTrueTb4b4b4(Te6edf3sawDbATb4b4b4, Ta5d6ff"Ta5d6ffcollector observed DB A's session before the switchTa5d6ff"Tb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff0Tb4b4b4, Te6edf3relatchedTb4b4b4.Te6edf3sizeTb4b4b4, Ta5d6ff"Ta5d6ffthe same collector re-latched onto the new (empty) DBTa5d6ff"Tb4b4b4)

Tb4b4b4}

T8b949e/** Minimal [DatabaseProvider] whose active DB the test can swap, mirroring a device/DB switch. */

Tf0883e@OptInTb4b4b4(Te6edf3ExperimentalCoroutinesApiTff7b72::Te6edf3classTb4b4b4)

Tff7b72private Tff7b72class T56d364TestProviderTb4b4b4(Te6edf3dbTb4b4b4: Te6edf3MeshtasticDatabaseTb4b4b4) Tb4b4b4: Te6edf3DatabaseProvider Tb4b4b4{

Tff7b72private Tff7b72val Te6edf3_currentDb Tff7b72= Te6edf3MutableStateFlowTb4b4b4(Te6edf3dbTb4b4b4)

Tff7b72private Tff7b72var Te6edf3writesAvailable Tff7b72= Tff7b72true

Tff7b72override Tff7b72val Te6edf3currentDbTb4b4b4: Te6edf3StateFlowTff7b72<Te6edf3MeshtasticDatabaseTff7b72> Tff7b72= Te6edf3_currentDb

Tff7b72override Tff7b72fun Tff7b72<Te6edf3TTff7b72> Td2a8ffobserveCurrentDbTb4b4b4(Te6edf3queryTb4b4b4: Tb4b4b4(Te6edf3MeshtasticDatabaseTb4b4b4) Tff7b72-Tff7b72> Te6edf3FlowTff7b72<Te6edf3TTff7b72>Tb4b4b4)Tb4b4b4: Te6edf3FlowTff7b72<Te6edf3TTff7b72> Tff7b72=

Te6edf3currentDbTb4b4b4.Te6edf3flatMapLatestTb4b4b4(Te6edf3queryTb4b4b4)

Tff7b72override Tff7b72suspend Tff7b72fun Tff7b72<Te6edf3TTff7b72> Td2a8ffwithReadDbTb4b4b4(Te6edf3blockTb4b4b4: Te6edf3suspend Tb4b4b4(Te6edf3MeshtasticDatabaseTb4b4b4) Tff7b72-Tff7b72> Te6edf3TTb4b4b4)Tb4b4b4: Te6edf3T Tff7b72= Te6edf3blockTb4b4b4(Te6edf3_currentDbTb4b4b4.Te6edf3valueTb4b4b4)

Tff7b72override Tff7b72suspend Tff7b72fun Tff7b72<Te6edf3TTff7b72> Td2a8ffwithDbTb4b4b4(Te6edf3blockTb4b4b4: Te6edf3suspend Tb4b4b4(Te6edf3MeshtasticDatabaseTb4b4b4) Tff7b72-Tff7b72> Te6edf3TTb4b4b4)Tb4b4b4: Te6edf3T? Tff7b72=

Tff7b72if Tb4b4b4(Te6edf3writesAvailableTb4b4b4) Te6edf3blockTb4b4b4(Te6edf3_currentDbTb4b4b4.Te6edf3valueTb4b4b4) Tff7b72else Tff7b72null

Tff7b72fun Td2a8ffsetWritesAvailableTb4b4b4(Te6edf3availableTb4b4b4: Tffa657BooleanTb4b4b4) Tb4b4b4{

Te6edf3writesAvailable Tff7b72= Te6edf3available

Tb4b4b4}

Tff7b72fun Td2a8ffswitchToTb4b4b4(Te6edf3dbTb4b4b4: Te6edf3MeshtasticDatabaseTb4b4b4) Tb4b4b4{

Te6edf3_currentDbTb4b4b4.Te6edf3value Tff7b72= Te6edf3db

Tb4b4b4}

Tb4b4b4}

Tb4b4b4}